package com.lxw.nonghe;

import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.config.OutputFile;
import com.baomidou.mybatisplus.generator.config.rules.DateType;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;

import java.util.Collections;

public class CodeGenerator {
    public static void main(String[] args) {
        try {
            // 数据库连接信息
            String url = "jdbc:mysql://localhost:3308/nonghe_db2";
            String username = "root";
            String password = "123456";

            FastAutoGenerator.create(url, username, password)
                    // 全局配置
                    .globalConfig(builder -> {
                        builder.author("lxw") // 设置作者
                                .outputDir("D:\\软件工程期末项目\\nonghe\\src\\main\\java") // 指定输出目录
                                .commentDate("yyyy-MM-dd") // 设置注释日期格式
                                .dateType(DateType.ONLY_DATE) // 设置日期类型
                                .enableSwagger() // 开启 swagger 模式
                                .disableOpenDir(); // 禁止打开输出目录
                    })
                    // 包配置
                    .packageConfig(builder -> {
                        builder.parent("com.lxw.nonghe") // 设置父包名
                                .entity("entity") // 设置实体类包名
                                .service("service") // 设置服务接口包名
                                .serviceImpl("service.impl") // 设置服务实现类包名
                                .mapper("mapper") // 设置 mapper 接口包名
                                .xml("mapper.xml") // 设置 mapper xml 包名
                                .controller("controller") // 设置控制器包名
                                .pathInfo(Collections.singletonMap(OutputFile.xml, "D:\\软件工程期末项目\\nonghe\\src\\main\\resources\\mapper")); // 设置 mapperXml 生成路径
                    })
                    // 策略配置
                    .strategyConfig(builder -> {
                        builder.addInclude("t_family_payment") // 设置需要生成的表名
                                .addTablePrefix("t_", "c_") // 设置过滤表前缀
                                .entityBuilder()
                                .enableLombok() // 开启 Lombok 模式
                                .enableChainModel() // 开启链式模型
                                .controllerBuilder()
                                .enableRestStyle() // 开启 Restful 风格
                                .serviceBuilder()
                                .formatServiceFileName("%sService") // 设置服务接口文件名格式
                                .formatServiceImplFileName("%sServiceImpl") // 设置服务实现类文件名格式
                                .mapperBuilder()
                                .enableMapperAnnotation() // 开启 @Mapper 注解
                                .formatMapperFileName("%sMapper") // 设置 mapper 接口文件名格式
                                .formatXmlFileName("%sMapper")
                                .enableFileOverride(); // 开启文件覆盖，在这里调用
                    })
                    // 模板引擎配置
                    .templateEngine(new FreemarkerTemplateEngine()) // 使用 Freemarker 引擎模板，默认的是 Velocity 引擎模板
                    .execute();
        } catch (Exception e) {
            System.err.println("代码生成过程中出现异常: " + e.getMessage());
            e.printStackTrace();
        }
    }
}
